# Interpolation methods for two dimensions.

licenses(["notice"])

package(default_visibility = ["//tf_quant_finance:__subpackages__"])

py_library(
    name = "interpolation_2d",
    srcs = ["__init__.py"],
    deps = [
        ":interpolation",
        # tensorflow dep,
    ],
)

py_library(
    name = "interpolation",
    srcs = ["interpolation_2d.py"],
    deps = [
        "//tf_quant_finance/math:piecewise",
        "//tf_quant_finance/math/interpolation/cubic",
        # tensorflow dep,
    ],
)

py_test(
    name = "interpolation_2d_test",
    size = "medium",
    srcs = ["interpolation_2d_test.py"],
    python_version = "PY3",
    shard_count = 3,
    deps = [
        "//tf_quant_finance",
        # test util,
        # numpy dep,
        # tensorflow dep,
    ],
)
